from pydantic_settings import BaseSettings

class Settings(BaseSettings):
    """项目基本配置类"""

    # ollama大模型配置
    OLLAMA_URL: str = "http://127.0.0.1:11434/v1"
    MODEL_NAME: str = "qwen2.5:7b"
    OLLAMA_API_KEY: str = "ollama"

    # neo4j数据库配置
    GRAPH_DB_URL:str = "bolt://localhost:7687"
    GRAPH_DB_AUTH:dict = {
            "username": "neo4j",
            "password": "12345678"
        }


settings = Settings()